package com.geek.video.utils;

import com.geek.video.enums.ResponseEnum;
import com.geek.video.service.dto.UserInfoDTO;
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import java.util.Date;
import java.util.HashMap;
import java.util.Map;

/**
 * Token生成工具类
 *
 * @author Steven
 */
@Component
public class JwtTokenUtil {
    /**
     * 密钥
     */
    @Value("${jwt.secret}")
    private String secret;

    /**
     * token过期时间
     */
    @Value("${jwt.expirationDate}")
    private Long expirationDate;

    /**
     * ⼿机唯⼀标识字段
     */
    @Value("${jwt.identifying}")
    private String identifying;

    /**
     * 从令牌中获取数据声明
     *
     * @param token 令牌
     * @return 数据声明
     */
    public Claims getClaimsFromToken(String token) {

        try {
            return Jwts.parser().setSigningKey(secret).parseClaimsJws(token).getBody();
        } catch (Exception e) {
            throw new RuntimeException(ResponseEnum.BASIC_TOKEN_EMPTY.getMessage());
        }
    }

    /**
     * 生成令牌
     * <p>
     * TIPS：根据自己业务需求在claims中添加不同的数据
     *
     * @param userInfo 用户信息
     * @param imei     手机标示
     * @return String             令牌
     */
    public String generateToken(UserInfoDTO userInfo, String imei) {

        Map<String, Object> claims = new HashMap<>(4);
        claims.put("sub", userInfo.getUserName());
        claims.put("userCode", userInfo.getUserCode());
        claims.put(identifying, imei);
        claims.put("created", new Date());

        return generateToken(claims);
    }


    /**
     * 从数据声明生成令牌
     *
     * @param claims 数据声明
     * @return String                   令牌
     */
    private String generateToken(Map<String, Object> claims) {

        // 设置token的有效期
        long expirationTime = System.currentTimeMillis() + expirationDate;

        return Jwts.builder().setClaims(claims).setExpiration(new Date(expirationTime))
                .signWith(SignatureAlgorithm.HS256, secret).compact();
    }


    /**
     * 判断令牌是否过期
     *
     * @param token 令牌
     * @return Boolean                  是否过期，false :未过期，true 过期
     */
    public Boolean isTokenExpired(String token) {

        Claims claims = getClaimsFromToken(token);
        Date expiration = claims.getExpiration();

        return expiration.before(new Date());
    }

    /**
     * 刷新令牌
     * <p>
     * 如果令牌有效性机制是动态的，就可以使用此方式延长令牌有效性
     *
     * @param token 原令牌
     * @return String                   新令牌
     */
    public String refreshToken(String token) {

        // 1. 判断令牌是否有效，如果token已经无效了那么就不能刷新
        if (isTokenExpired(token)) {
            throw new RuntimeException(ResponseEnum.BASIC_TOKEN_EMPTY.getMessage());
        }
        // 2. 生成新令牌
        Claims claims = getClaimsFromToken(token);
        claims.put("created", new Date());

        return generateToken(claims);
    }

}
